/*
 * Copyright (C) 2023 HiHope Open Source Organization .
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

 #include <stdio.h>
 #include <unistd.h>
 
 #include "ohos_init.h"
 #include "cmsis_os2.h"
 #include "iot_gpio.h"
 #include "hi_io.h"
 #include "iot_pwm.h"
 
 #define LED_TASK_GPIO    9
 #define IOT_GPIO_KEY     5 // 14 
 #define DELAY_TICKS_30  (30)
 #define LED_TASK_STACK_SIZE 512
 #define LED_TASK_PRIO   (25)
 #define GPIO_HIGH       (1)
 #define GPIO_LOW        (0)
 #define RED_LED_PIN_NAME 10 //2
#define GREEN_LED_PIN_NAME 11  //7
#define BLUE_LED_PIN_NAME 12  //8
#define IOT_PWM_DUTY_MAX 100
#define PWM_FREQ_DIVITION 64000

 enum LedState {
     RED = 0,
     GREEN,
     BLUE,
 };
 
 enum LedState g_ledState = RED;
 static int g_count = 100;
 
 static void *LedTask(void)
 {
     while (g_count) {
         switch (g_ledState) {
             case RED:
                 for (int i = 1; i < IOT_PWM_DUTY_MAX; i += 1) {
                    IoTPwmStart(1, i, PWM_FREQ_DIVITION);
                    usleep(25000);
                    IoTPwmStop(1);
                    }
    
                 break;
             case GREEN:
                for (int i = 1; i < IOT_PWM_DUTY_MAX; i += 1) {
                IoTPwmStart(2, i, PWM_FREQ_DIVITION);
                usleep(25000);
                IoTPwmStop(2);
                }
                
                 break;
             case BLUE:
                for (int i = 1; i < IOT_PWM_DUTY_MAX; i += 1) {
                IoTPwmStart(3, i, PWM_FREQ_DIVITION);
                usleep(25000);
                IoTPwmStop(3);
                }
                
                 break;
             default:
                 osDelay(DELAY_TICKS_30);
                 break;
         }
     }
 
     return NULL;
 }
 
 static void OnButtonPressed(void)
 {
     enum LedState nextState;
     switch (g_ledState) {
         case RED:
             nextState = GREEN;
             break;
         case GREEN:
             nextState = BLUE;
             break;
         case BLUE:
             nextState = RED;
             break;
         default:
             break;
     }
 
     g_ledState = nextState;
     g_count--;
 }
 
 static void LedExampleEntry(void)
 {
     osThreadAttr_t attr;
 
     IoTGpioInit(RED_LED_PIN_NAME);
    IoTGpioInit(GREEN_LED_PIN_NAME);
    IoTGpioInit(BLUE_LED_PIN_NAME);
  
     IoTGpioInit(IOT_GPIO_KEY);
     hi_io_set_func(IOT_GPIO_KEY, HI_IO_FUNC_GPIO_5_GPIO);
     IoTGpioSetDir(IOT_GPIO_KEY, IOT_GPIO_DIR_IN);
     hi_io_set_pull(IOT_GPIO_KEY, GPIO_HIGH);

     hi_io_set_func(RED_LED_PIN_NAME, HI_IO_FUNC_GPIO_10_PWM1_OUT);
     hi_io_set_func(GREEN_LED_PIN_NAME, HI_IO_FUNC_GPIO_11_PWM2_OUT);
     hi_io_set_func(BLUE_LED_PIN_NAME, HI_IO_FUNC_GPIO_12_PWM3_OUT);
     IoTPwmInit(3);
     IoTPwmInit(1);
     IoTPwmInit(2);
     IoTGpioRegisterIsrFunc(IOT_GPIO_KEY, IOT_INT_TYPE_EDGE,
         IOT_GPIO_EDGE_FALL_LEVEL_LOW,
         OnButtonPressed, NULL);
 
     attr.name = "LedTask";
     attr.attr_bits = 0U;
     attr.cb_mem = NULL;
     attr.cb_size = 0U;
     attr.stack_mem = NULL;
     attr.stack_size = LED_TASK_STACK_SIZE;
     attr.priority = LED_TASK_PRIO;
 
     if (osThreadNew((osThreadFunc_t)LedTask, NULL, &attr) == NULL) {
         printf("[LedExample] Falied to create LedTask!\n");
     }
 }
 SYS_RUN(LedExampleEntry);